package hash.hashcode;
/**
 * 
 * @Title: CandyHashTable.java 
 * @Package hash.hashcode 
 * @Description: 自定义哈希表
 * @author CandyWall   
 * @date 2021年1月12日 下午7:00:40 
 * @version V1.0
 */

import java.util.TreeMap;

public class CandyHashTable_v1<K, V> {
    private TreeMap<K, V>[] hashTable;
    private int M;      // 哈希表容量
    private int size;   // 存储元素的个数
    public CandyHashTable_v1(int M) {
        this.M = M;
        hashTable = new TreeMap[M];
        for(int i = 0; i < M; i++) {
            hashTable[i] = new TreeMap<>();
        }
        size = 0;
    }
    
    public CandyHashTable_v1() {
        this(97);
    }
    
    private int hash(K key) {
        //               16进制转2进制 
        // 0x7fffffff ----------------> 01111111 11111111 11111111 11111111
        // 一个整型的数占4个字节，总共是32位，其中最高位是符号位，0表示正数，1表示负数，
        // 让key的hashCode()值与上0x7fffffff可以去掉符号位
        return (key.hashCode() & 0x7fffffff) % M;
    }
    
    public int getSize() {
        return size;
    }
    
    public boolean isEmpty() {
        return size == 0;
    }
    
    public void add(K key, V value) {
        TreeMap<K, V> map = hashTable[hash(key)];
        if(!map.containsKey(key))
            size++;
        map.put(key, value);
    }
    
    public V remove(K key) {
        TreeMap<K, V> map = hashTable[hash(key)];
        V ret = null;
        if(map.containsKey(key)) {
            ret = map.remove(key);
            size--;
        }
        return ret;
    }
    
    public void set(K key, V value) {
        TreeMap<K, V> map = hashTable[hash(key)];
        if(!map.containsKey(key)) {
            throw new IllegalArgumentException("key:" + key + " 不存在！");
        }
        map.put(key, value);
    }
    
    public boolean contains(K key) {
        return hashTable[hash(key)].containsKey(key);
    }
    
    public V get(K key) {
        return hashTable[hash(key)].get(key);
    }
}
